from codex import *

schedule = ["Computer Science", "English", "Biology", 
            "Algebra 2", "Band", "Health"]

def check_schedule(course):
    if course in schedule:
        display.print(course)
        display.print("On the schedule")
    else:
        display.print(course)
        display.print("Not on schedule")

# === Main Program ===
while True:
    course = input("Enter a class:")
    check_schedule(course)